# 雇员 　编写一个名为Employee 的类，其方法
# __init__() 接受名、姓和年薪，并将它们存储在属性中。编
# 写一个名为give_raise() 的方法，它默认将年薪增加5000美
# 元，但也能够接受其他的年薪增加量。
# 为Employee 编写一个测试用例，其中包含两个测试方
# 法：test_give_default_raise() 和
# test_give_custom_raise() 。使用方法setUp() ，以免在
# 每个测试方法中都新建雇员实例。运行这个测试用例，确认两
# 个测试都通过了。

# test_employee.py

import unittest
from employee import Employee

class TestEmployee(unittest.TestCase):
	"""针对Employee类的测试"""

	def setUp(self):
		"""创建一个雇员对象，供使用的测试方法使用"""

		self.employee = Employee('Hongyi', 'Wu', 250000)		# 创建待测试类的实例

	def test_give_default_raise(self):
		"""测试雇员年薪是否按默认值增加"""
		new_salary = self.employee.give_raise()
		self.assertEqual(new_salary, self.employee.salary)

	def test_give_custom_raise(self):
		"""测试雇员年薪是否按指定值增加"""
		new_salary = self.employee.give_raise(10000)
		self.assertEqual(new_salary, self.employee.salary)


if __name__ == '__main__':
	unittest.main()

	